import sys
from pathlib import Path

base_path = str(Path(__file__).resolve().parent.parent)
sys.path.append(base_path)
from ai_configs import defalut_ai_config
from main import ai_chat_stream, default_on_receiving

_cot = "请一步一步思考，得出结论"

_system_content = f"""
# 角色
你是一名专业的质检人员，给定一段用户与手机号码流量套餐客服的对话，你的任务是判断客服回答是否符合规范

# 思考过程
{_cot}

# 语言
与用户提问的语言保持一致

# 客服回答规范评判标准
1.必须有礼貌
2.必须使用官方口语，不能使用网络用语，不能使用网络用语，不能使用网络用语
3.介绍套餐时，必须准确介绍套餐信息，上述套餐信息有缺失或者与事实不符，都是不符合规范的

# 已知的手机号码流量套餐
1. 经济套餐：月费9.9元，月流量10G，免费通话30分钟；
2. 畅游套餐：月费29.9元，月流量300G，免费通话100分钟；
3. 无限套餐：月费198元，月流量无限，免费通话300分钟；
4. 校园套餐：月费19.9,元，月流量200G，免费通话30分钟，限在校学生办理；
"""

_user_content = """
用户：你们有什么流量大的套餐
客服：亲，你好，我们现在正在推广无限套餐，每月198元就可以享受月流量无限服务，免费通话300分钟，请问您感兴趣吗？
"""

ai_chat_stream(
    ai_config=defalut_ai_config["deepseek"],
    on_receiving=default_on_receiving,
    histories=[
        {"role": "system", "content": _system_content},
        {"role": "user", "content": _user_content},
    ],
)
